package online.yangcloud.common.aspect;

import cn.hutool.core.annotation.AnnotationUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import online.yangcloud.common.annotation.RepeatRequest;
import online.yangcloud.common.annotation.SessionValid;
import online.yangcloud.common.annotation.TokenEntity;
import online.yangcloud.common.common.AppConstants;
import online.yangcloud.common.tools.ExceptionTools;
import online.yangcloud.common.tools.RedisTools;
import online.yangcloud.common.tools.SystemTools;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * @author zhuboyang
 * @since 2022/12/31 11:40
 */
@Aspect
@Component
public class ServletLogAspect {

    public static final Logger logger = LoggerFactory.getLogger(ServletLogAspect.class);

    @Resource
    private ApplicationContext applicationContext;

    @Resource
    private RedisTools redisTools;

    /**
     * AOP 通知：
     * 1. 前置通知：在方法调用之前执行
     * 2. 后置通知：在方法正常调用之后执行
     * 3. 环绕通知：在方法调用之前和之后，都分别可以执行的通知
     * 4. 异常通知：如果在方法调用过程中发生异常，则通知
     * 5. 最终通知：在方法调用之后执行
     * 切面表达式：
     * execution 代表所要执行的表达式主体
     * 第一处 * 代表方法返回类型 * 代表所有类型
     * 第二处 包名代表 aop 监控的类所在的包
     * 第三处 .. 代表该包以及其子包下的所有类方法
     * 第四处 * 代表类名，* 代表所有类
     * 第五处 *(..) * 代表类中的方法名 ( 所有方法名 )，() 表示方法的参数，.. 表示可以接受任何参数
     */
    @Around("@annotation(online.yangcloud.common.annotation.TimeConsuming)")
    public Object recordTimeLog(ProceedingJoinPoint joinPoint) throws Throwable {
        Class<?> clazz = joinPoint.getTarget().getClass();
        String methodName = joinPoint.getSignature().getName();

        // 获取执行的类名和方法名
        logger.info("====== 开始执行 {}.{} ======", clazz, methodName);

        // 记录开始执行方法的时间
        long begin = System.currentTimeMillis();

        // 执行目标 service 方法
        Object result = joinPoint.proceed();

        // 记录结束时间
        long end = System.currentTimeMillis();
        long takeTime = end - begin;

        if (takeTime >= 0 && takeTime < AppConstants.Consuming.NORMAL_TIMING) {
            logger.info("====== {}.{} 执行结束，耗时：{} 毫秒 =======", clazz, methodName, takeTime);
        } else if (takeTime >= AppConstants.Consuming.NORMAL_TIMING && takeTime < AppConstants.Consuming.WARN_TIMING) {
            logger.warn("====== {}.{} 执行结束，耗时：{} 毫秒 =======", clazz, methodName, takeTime);
        } else {
            logger.error("====== {}.{} 执行结束，耗时：{} 毫秒 =======", clazz, methodName, takeTime);
        }

        return result;
    }

    @Around("execution(* online.yangcloud.*.controller.*.*(..)) ")
    public Object handleControllerMethod(ProceedingJoinPoint joinPoint) throws Throwable {
        // 具体请求的类和方法
        Class<?> clazz = joinPoint.getTarget().getClass();
        Class<?>[] parameterTypes = ((MethodSignature) joinPoint.getSignature()).getMethod().getParameterTypes();
        Method method = clazz.getMethod(joinPoint.getSignature().getName(), parameterTypes);
        logger.info("is requesting [{}.{}()]", clazz.getName(), method.getName());

        // 检测是否需要控制请求量
        if (AnnotationUtil.hasAnnotation(method, RepeatRequest.class)) {
            preventRepeatRequest();
        }

        // 检测是否需要回写用户信息和检测登录态
        Object[] args = joinPoint.getArgs();
        SessionValid valid = AnnotationUtil.getAnnotation(method, SessionValid.class);
        if (ObjUtil.isNotNull(valid)) {
            if (valid.needValid()) {
                rewriteSignInStatusInformation(args);
            }
        }
        return joinPoint.proceed(args);
    }

    /**
     * 阻止重复请求
     */
    public void preventRepeatRequest() {
        String key = "repeat_request";
        Boolean flag = redisTools.setIfAbsent(key, StrUtil.EMPTY);
        if (!flag) {
            ExceptionTools.businessLogger("操作太快了，系统反应不过来了");
        }
    }

    /**
     * 将登录态信息回写到请求的参数中
     *
     * @param args 请求参数
     */
    private void rewriteSignInStatusInformation(Object[] args) {
        // 获取到登录态附属的类
        Map<String, Object> map = applicationContext.getBeansWithAnnotation(TokenEntity.class);
        if (map.isEmpty()) {
            ExceptionTools.businessLogger("未检测到带有登录态附属类");
        }
        List<String> keys = new ArrayList<>(map.keySet());
        Object bean = map.get(keys.get(0));

        // 将登录信息进行回写
        Object obj = null;
        int index = -1;
        if (args != null && args.length > 0) {
            for (int i = 0; i < args.length; i++) {
                if (args[i].getClass().equals(bean.getClass())) {
                    String sessionId = SystemTools.getHeaders().getAuthorization();
                    if (StrUtil.isBlank(sessionId)) {
                        ExceptionTools.paramLogger();
                    } else {
                        String tokenInfo = redisTools.get(AppConstants.Account.LOGIN_TOKEN + sessionId);
                        if (StrUtil.isBlank(tokenInfo)) {
                            ExceptionTools.businessLogger();
                        }
                        obj = JSONUtil.toBean(tokenInfo, bean.getClass());
                    }
                    index = i;
                    break;
                }
            }
        }
        if (index != -1) {
            args[index] = obj;
        }
    }

}
